hackerrank python 3